import os

from langchain_openai import ChatOpenAI


def getCommonLLM(streaming: bool = False, callbacks=None):
    # 创建LLM
    llm = ChatOpenAI(
        api_key=os.getenv("DEEPSEEK_API_KEY"),
        base_url="https://api.deepseek.com/v1",
        model="deepseek-chat",
        streaming=streaming,
    )
    return llm


def getThinkingLLM(streaming: bool = False, callbacks=None):
    # 创建LLM
    llm = ChatOpenAI(
        api_key=os.getenv("DEEPSEEK_API_KEY"),
        base_url="https://api.deepseek.com/v1",
        model="deepseek-reasoner",
        streaming=streaming,
    )
    return llm